from dask import delayed

# Define functions
@delayed
def square(x):
    return x ** 2

@delayed
def add(x, y):
    return x + y

# Build computation graph
a = square(2)
b = square(3)
result = add(a, b)

# Compute result
print("Result:", result.compute())